#ifndef _MODEL_TRANS_TRANSFERTOHOSTMEMORY_
#define _MODEL_TRANS_TRANSFERTOHOSTMEMORY_

#include "../../../function.h"

namespace sp::Model::Trans{

void registe_transferToHostMemory();

class transferToHostMemory: public PythonFunction, public GpuFunction{
public:
    transferToHostMemory(std::vector<std::string> v);
    transferToHostMemory();
    virtual ~transferToHostMemory();
    static void registe();
    static Function* createObject(std::vector<std::string> params);
    static Function* createObject2();
    virtual void start();
    virtual bool waitForResource();
    virtual void process(std::vector<std::vector<Data*>>& data_input, std::vector<Data*>& data_output);
    virtual bool releaseResource();
    virtual void finish();
    virtual void copy(Function* other);
};

};

#endif